/*
 * sysmo 服务器监控图表类
 * Copyright (C) 2009  YuLicheng
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.sysmo.server.node.item.graphgroup;

import java.io.OutputStream;
import java.sql.SQLException;
import java.util.Date;
import java.util.List;
import java.util.Vector;
import org.sysmo.db.DatabaseInterface;
import org.sysmo.db.DbGraph;
import org.sysmo.db.DbValue;
import org.sysmo.db.DbWarning;
import org.sysmo.server.graph.Graph;
import org.sysmo.server.graph.Point;
import org.sysmo.server.node.item.Item;
import org.sysmo.server.node.item.ReceivedResult;
import org.sysmo.server.node.item.Result;
import org.sysmo.server.node.item.Warning;
import org.sysmo.server.node.item.graphgroup.value.Value;
import org.sysmo.server.node.item.graphgroup.value.ValueExistsException;
import org.sysmo.server.util.InstanceNotInitializedException;

/**
 *
 * 监控项中的图表对象。一个图表中可包含多个监控值。由于对图表的元数据修改不需要通知
 * 客户端，而且监控数据的传送也是依靠监控项传递的，因此不需要与客户端进行通讯。调用
 * 各种set类方法对此对象实例的参数进行修改后，必须显式调用{@link #sync()}方法将所
 * 有修改一次性写入数据库。
 *
 * @author YuLicheng
 * @see Item
 */
public class GraphGroup {

    /**
     * 指定图表类型的枚举。图表类型是各监控值记录在同一图表上的呈现方式。
     * 
     * @see #getGraphType()
     * @see #setGraphType(GraphType)
     */
    public enum GraphType {
        /**
         * 普通类型（折线图）。
         */
        NORMAL,
        /**
         * 面积叠加图。
         */
        AREA;
    };

    /**
     * 获取图表时指定的图表数据粒度。
     *
     * @see #getGraph(int, int, int)
     */
    public enum GType {
        /**
         * 显示间隔在1秒钟到59秒钟之间的数据
         */
        G_BY_SECOND,
        /**
         * 显示间隔在1分钟到59分钟之间的数据
         */
        G_BY_MINUTE,
        /**
         * 显示间隔在1小时到1天之内的数据
         */
        G_BY_HOUR,
        /**
         * 显示间隔在1天到1周之内的数据
         */
        G_BY_DAY,
        /**
         * 显示间隔在1周到1个月之内的数据
         */
        G_BY_WEEK,
        /**
         * 显示间隔在1个月到1年的数据
         */
        G_BY_MONTH,
        /**
         * 显示间隔1年或以上的数据
         */
        G_BY_YEAR,
    };

    private DatabaseInterface dbInterface;
    private DbGraph graph;

    /**
     * 实例化数据库中已存在的监控图表。
     *
     * @param dbInterface 数据库访问接口。
     * @param id 已存在的监控项ID。
     * @throws GraphGroupNotExistsException 图表ID不存在。
     */
    public GraphGroup(DatabaseInterface dbInterface, int id)
            throws GraphGroupNotExistsException {
        this.dbInterface = dbInterface;
        try {
            graph = dbInterface.query_GraphTable(id);
        } catch (Exception e) {
            e.printStackTrace();
        }
        if (graph == null) {
            throw new GraphGroupNotExistsException(id);
        }
    }

    /**
     * 创建一个新的监控图表。
     *
     * @param dbInterface 数据库访问接口。
     * @param name 新监控项名称。
     * @param itemID 所属监控项ID。
     * @throws GraphGroupExistsException 图表名称已存在。
     */
    public GraphGroup(DatabaseInterface dbInterface, String name, int itemID)
            throws GraphGroupExistsException {
        this.dbInterface = dbInterface;
        graph = new DbGraph();
        //TODO: 为DbGraph新对象指定ID。
        graph.gid = 0;
        graph.iid = itemID;
        graph.name = name;
        graph.type = GraphType.NORMAL.ordinal();
        try {
            if (dbInterface.query_GraphTable(name, itemID) != null) {
                throw new GraphGroupExistsException(name);
            }
            dbInterface.update_GraphTable(graph);
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }

    /**
     * 返回图表名称。
     *
     * @return 图表名称。
     */
    public String getName() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        return graph.name;
    }

    /**
     * 设置图表名称。
     *
     * @param name 图表新名称。
     * @throws GraphGroupExistsException 存在同名图表。
     */
    public void setName(String name) throws GraphGroupExistsException {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        if (!name.equals(graph.name)) {
            try {
                if (dbInterface.query_GraphTable(name, graph.iid) != null) {
                    throw new GraphGroupExistsException(name);
                }
                graph.name = name;
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 返回此图表ID。
     *
     * @return 图表ID。
     */
    public int getID() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        return graph.gid;
    }

    /**
     * 获得图表所属的监控项ID。
     *
     * @return 图表所属监控项ID。
     */
    public int getParentID() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        return graph.iid;
    }

    /**
     * 返回此图表类型。
     *
     * @return 图表类型。
     */
    public GraphType getGraphType() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        return GraphType.values()[graph.type];
    }

    /**
     * 更改图表类型。
     *
     * @param graphType 新图表类型。
     * @see GraphType
     */
    public void setGraphType(GraphType graphType) {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        graph.type = graphType.ordinal();
    }

    /**
     * 在当前图表中增加新的监控值。
     * 
     * @param name 新监控值名称。
     * @return 监控值对象。
     * @throws ValueExistsException 存在同名监控值。
     */
    public Value addValue(String name) throws ValueExistsException {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        return new Value(dbInterface, name, graph.gid);
    }

    /**
     * 返回所有此图表中的监控值对象组成的列表。
     *
     * @return 监控值对象列表。
     */
    public List<Value> getAllValues() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        Vector<Value> allValues = new Vector<Value>();
        try {
            Vector<DbValue> values = dbInterface.query_ValueTableOfGraph(graph.gid);
            for (DbValue value : values) {
                try {
                    allValues.add(new Value(dbInterface, value.vid));
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return allValues;
    }

    /**
     * 获得此图表的所有警报列表。
     *
     * @return 所有警告列表。
     */
    public List<Warning> getAllWarnings() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        //TODO: 查询警告的默认无关ID
        Vector<DbWarning> warnings;
        try {
            warnings = dbInterface.query_WarningTable(0, 0, graph.gid, 0);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
        Vector<Warning> allWarnings = new Vector<Warning>();
        for (DbWarning warning : warnings) {
            allWarnings.add(new Warning(dbInterface, warning));
        }
        return allWarnings;
    }

    /**
     * 保存客户端发送的监控数据。此方法会将相应的数据存入数据库。
     * <br />
     *
     * 此方法会忽略非本图表的记录，并调用相应的监控值的saveData方法来存储。
     *
     * @param results 监控结果集合。
     */
    public void saveData(List<ReceivedResult> results) {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        Vector<ReceivedResult> thisResults = new Vector<ReceivedResult>();
        for (ReceivedResult res : results) {
            if (res.getGraphName().equals(graph.name)) {
                Warning warning = res.getWarning();
                if (warning != null) {
                    warning.setGraphID(graph.gid);
                }
                thisResults.add(res);
            }
        }
        List<Value> allValues = getAllValues();
        for (Value value : allValues) {
            value.saveData(thisResults);
        }
    }

    /**
     * 图表中最多同时显示的点数
     */
    public static final int MAX_POINTS = 80;

    /**
     * 按时间获取图表。此方法会用指定时间段内的数据调用图表绘制模块绘制图表，并且将
     * 相应的JPEG图片写入参数中指定的输出流。
     *
     * @param startTime 图表中数据的开始时间。
     * @param endTime 图表中数据结束时间。
     * @param g 数据显示粒度。
     * @param out 图表图片的输出流（JPEG）。
     *
     * @see GType
     */
    public void getGraph(Date startTime, Date endTime, GType g, OutputStream out) {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        int lowLayer = 0, highLayer = Integer.MAX_VALUE;
        /*switch (g) {
            case G_BY_SECOND: lowLayer = 0; highLayer = 59; break;
            case G_BY_MINUTE: lowLayer = 60; highLayer = 60 * 60 - 1; break;
            case G_BY_HOUR: lowLayer = 60 * 60; highLayer = 60 * 60 * 24 - 1; break;
            case G_BY_DAY: lowLayer = 60 * 60 * 24; highLayer = 60 * 60 * 24 * 7 - 1; break;
            case G_BY_WEEK: lowLayer = 60 * 60 * 24 * 7; highLayer = 60 * 60 * 24 * 7 * 4 - 1; break;
            case G_BY_MONTH: lowLayer = 60 * 60 * 24 * 7 * 4; highLayer = 60 * 60 * 24 * 7 * 4 * 12 - 1; break;
            case G_BY_YEAR: lowLayer = 60 * 60 * 24 * 7 * 4 * 12; highLayer = Integer.MAX_VALUE; break;
        }*/
        Vector<Vector<Point>> graphPoints = new Vector<Vector<Point>>();
        Vector<String> valueNames = new Vector<String>();
        Vector<Integer> derMode = new Vector<Integer>();
        List<Value> allValues = getAllValues();
        int max = 0;
        Vector<Point> maxKey = null;
        for (Value value : allValues) {
            Vector<Point> valuePoints = new Vector<Point>();
            List<Result> allResults = value.getResults(startTime, endTime, lowLayer, highLayer);
            for (Result res : allResults) {
                valuePoints.add(new Point(res.getTime(), res.getValue()));
            }
            if (valuePoints.size() > max) {
                max = valuePoints.size();
                maxKey = valuePoints;
            }
            graphPoints.add(valuePoints);
            valueNames.add(value.getName());
            int dMode = 0;
            switch (value.getStatisticType()) {
                case SCALAR: dMode = 0; break;
                case DERIVATIVE: dMode = 1; break;
                case ACCUMULATED: dMode = 2; break;
            }
            derMode.add(dMode);
        }
        for (Vector<Point> points : graphPoints) {
            int diff = max - points.size();
            for (int i = 0; i < diff; i++) {
                points.insertElementAt(new Point(maxKey.get(i).getTime(), 0), i);
            }
        }
        if (graphPoints.size() == 0) {
            graphPoints.add(new Vector<Point>());
            derMode.add(0);
            valueNames.add("");
        } else if (maxKey != null && maxKey.size() > MAX_POINTS) { //压缩显示的点数
            Vector<Vector<Point>> newPoints = new Vector<Vector<Point>>();
            long lowBound = maxKey.firstElement().getTime().getTime();
            long upBound = maxKey.lastElement().getTime().getTime();
            long timeStep = (upBound - lowBound) / MAX_POINTS;
            int i = 0;
            int size = maxKey.size();
            for (int k = 0; k < graphPoints.size(); k++) {
                newPoints.add(new Vector<Point>());
            }
            for (long bound = lowBound + timeStep; bound <= upBound; bound += timeStep) {
                int j = i;
                while (j < size && maxKey.get(j).getTime().getTime() < bound) {
                    j++;
                }
                int len = j - i;
                if (len <= 0) {
                    continue;
                }
                for (int k = 0; k < graphPoints.size(); k++) {
                    double total = 0;
                    double ttime = 0;
                    Vector<Point> thisPoints = graphPoints.get(k);
                    for (int t = i; t < j; t++) {
                        Point thisPoint = thisPoints.get(t);
                        total += thisPoint.getValue();
                        ttime += thisPoint.getTime().getTime();
                    }
                    newPoints.get(k).add(new Point(new Date((long)(ttime / len)), (int)(total / len)));
                }
                i = j;
            }
            graphPoints = newPoints;
        }
        Graph graphImage = new Graph();
        int graphType = 0;
        switch (getGraphType()) {
            case NORMAL: graphType = 0; break;
            case AREA: graphType = 1; break;
        }
        graphImage.drawGragh(graphPoints, graphType, out, valueNames, derMode);
    }
    
    /**
     * 从数据库删除与此对象相关的所有数据并且将当前对象设为无效。
     */
    public void remove() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        try {
            dbInterface.delete_GraphTable(graph.gid);
        } catch (Exception e) {
            e.printStackTrace();
        }
        graph = null;
    }

    /**
     * 将此实例的元数据修改写入数据库。
     */
    public void sync() {
        if (graph == null) {
            throw new InstanceNotInitializedException();
        }
        try {
            dbInterface.update_GraphTable(graph);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}
